import matplotlib.pyplot as plt
import numpy as np

# =================== 散点图 =========================
x = np.linspace(0, 1, 20)  # (20,)
y = 3 * x + 2

# 噪点 ：值越大，点越散乱，值越小，越接近线性关系
noise = 0.4
# 加入随机值，让y变得散乱
y += np.random.normal(0, noise, y.shape)
plt.plot(x, y, 'ro')

# =================== 预测线 =========================
# 预测的w,b
w = 0.1
b = 0.1
# 预测的 predict_y

predict_y = w * x + b
plt.plot(x, predict_y, 'b--')
# =================== 距离衡量 =========================
distance = np.sum((predict_y - y) ** 2)
print(distance)

plt.show()
